/*
 * Fernanda Ibarrola - 0711920
 * Luiza Silva       - 0810905
 *
 */


#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "paciente.h"
#include "hash.h"
#include "heap.h"

#define MAX 65000

enum {
	CORACAO=0, FIGADO, RIM, CORNEA
};

Paciente** leArquivo (char* nomeArquivo, FILE* arquivoOut)
{
	int hashIndex;
	int temp;
	FILE* fCadastro;
	Paciente* p;
	Paciente** vetorPacientes;
	Heap* heaps[4];
	int maior, segMaior, heap1, heap2;

	for (temp=0; temp<4; temp++) {
		heaps[temp] = criarHeap(definePrioridade);
	}
	
	vetorPacientes = (Paciente**) criaVetor(MAX);
	
 	fCadastro = fopen(nomeArquivo, "r");
 	
 	if (!fCadastro) {
 		printf("Erro ao ler o arquivo.\n");
 		exit(1);
 	}
 	
 	p = (Paciente*)malloc(sizeof(Paciente));
 	
 	/* Aloca todas as informações no vetor auxiliar p. */
	while (fscanf(fCadastro, "%09d", &p->CPF) != EOF) {		
 		fscanf(fCadastro, "%d", &temp);
 		fscanf(fCadastro, "%s", p->nome);
 		fscanf(fCadastro, "%s", p->orgao);
 		fscanf(fCadastro, "%f", &p->gravidade);
 		fscanf(fCadastro, "%d", &p->tempoEspera);
 		fscanf(fCadastro, "%s", p->contato);
 		p->colisao = 0;
 		p->prox = NULL;
 		
 		/* Ulitiza a função hashFunc para achar o índice do paciente no vetor. */
 		hashIndex = hashFunc(p->CPF);

		/* Se a posição tiver vazia, apenas coloca o paciente no local. */
 		if (vetorPacientes[hashIndex] == NULL) {
 			vetorPacientes[hashIndex] = p;
 		}
 			
 		/* Se não estiver vazia, cria-se uma lista de pacientes com a cabeça no índice retornado pela hashFunc. */
		else {
			hashCol(vetorPacientes[hashIndex], p);
		}
	
		/* Agora, o heap */

		
		if (!strcmp(p->orgao,"CORACAO")) {
			adicionarElemento(heaps[CORACAO], p);
		} 
		else if (!strcmp(p->orgao,"RIM")) {
			adicionarElemento(heaps[RIM], p);
		}
		else if (!strcmp(p->orgao,"CORNEA")) {
			adicionarElemento(heaps[CORNEA], p);
		}
		else {
			adicionarElemento(heaps[FIGADO], p);
		}
		
		/* fim da parte do heap */
		
		p = (Paciente*)malloc(sizeof(Paciente));
 	} 	
 
	free(p); /* libera o que ele alocou a toa */
 	fclose(fCadastro);

	/* Imprime os dois maiores heaps */
	/* Primeiro, vamos achar os dois maiores */
	maior = -1; segMaior = -1;

	for (temp=0; temp<4; temp++) {
		if (maior<tamanhoHeap(heaps[temp])) {
			segMaior = maior;
			heap2 = heap1;
			maior = tamanhoHeap(heaps[temp]);
			heap1 = temp;
		} 
		else {
			if (segMaior<tamanhoHeap(heaps[temp])) {
				segMaior = tamanhoHeap(heaps[temp]);
				heap2 = temp;
			}
		}
	}

	/* Agora, vamos imprimi-los */
	fprintf(arquivoOut, "2. Estado inicial da fila de transplante - orgaos criticos\n\n");

	imprimePrimeiros(heaps[heap1], arquivoOut);
	imprimePrimeiros(heaps[heap2], arquivoOut);	

	/* Tira pacientes atendidos e muda as prioridades */
	for (temp=0; temp<4; temp++) {
		if (temp == heap1 || temp == heap2)
			atualizaFila(heaps[temp], tamanhoHeap(heaps[temp])/2);
		else
			atualizaFila(heaps[temp], 3);
	}
	
	maior = -1; segMaior = -1;

	for (temp=0; temp<4; temp++) {
		if (maior<tamanhoHeap(heaps[temp])) {
			segMaior = maior;
			heap2 = heap1;
			maior = tamanhoHeap(heaps[temp]);
			heap1 = temp;
		} 
		else {
			if (segMaior<tamanhoHeap(heaps[temp])) {
				segMaior = tamanhoHeap(heaps[temp]);
				heap2 = temp;
			}
		}
	}

	/* Agora, vamos imprimi-los de novo*/
	fprintf(arquivoOut, "3. Estado da fila de transplante apos 30 dias - orgaos criticos\n\n");
	
	imprimePrimeiros(heaps[heap1], arquivoOut);
	imprimePrimeiros(heaps[heap2], arquivoOut);

	for (temp=0; temp<4; temp++)
		destruirHeap(heaps[temp]);
	
	/* termina aqui a parte de heap */
 	
 	return vetorPacientes;
}


void zeraVetor (Paciente** hashVetor, int tam) 
{
	int i;

	for (i=0; i<tam; i++) {
		if (hashVetor[i]) {
			free(hashVetor[i]);
			hashVetor[i] = NULL;
		}
	}

	free(hashVetor);
}

int definePrioridade(void* pacienteA, void* pacienteB) {
	Paciente* a, *b;
	a = (Paciente*) pacienteA;
        b = (Paciente*) pacienteB;
	
	if (2*365*(a->gravidade) + (a->tempoEspera) > 2*365*(b->gravidade) + (b->tempoEspera))
		return 1; 
	else 
		return -1;
}

void imprimePrimeiros(Heap* h, FILE* arquivoOut) {
	Paciente* p[3]; /* Guarda provisoriamente os tres primeiros */
	int i=0;

	for (i=0; i<3; i++)
		p[i]=NULL;
	
	i=0;

	while (i<3 && !heapVazia(h)) { 
		p[i] = retornaTopo(h);
		tirarElemento(h);
		i++;
	}

	if (p[0]!=NULL) {
		fprintf(arquivoOut, "Orgao: %s\n\n", p[0]->orgao);
	}

	for (i=0; i<3; i++) {
		if (p[i]!=NULL)
			fprintf(arquivoOut, "CPF: %d\t gravidade: %f\t tempo de espera: %d\n", 
					p[i]->CPF, p[i]->gravidade, p[i]->tempoEspera);	
	}

	fprintf(arquivoOut, "\n");

	for (i=0; i<3; i++) {
		if (p[i]!=NULL)
			adicionarElemento(h, p[i]);
	}
}

void mudaPrioridade(Paciente* p) {
	if (p->gravidade > 0.8)
		p->gravidade*=0.8;
	else
		if (p->gravidade > 0.6 && p->gravidade <= 0.8)
			p->gravidade*=1.10;
}

void atualizaFila(Heap* h, int numAtendidos) {
	int i=0;
	int numRestantes = tamanhoHeap(h)-numAtendidos;
	Paciente** ps;
	
	if (numRestantes<0) {
		numAtendidos=tamanhoHeap(h);
		numRestantes=0;
	}	

	for (i=0; i<numAtendidos; i++)
		tirarElemento(h);

	if (!numRestantes)
		return;

	ps = malloc(sizeof(Paciente*)*numRestantes);

	if (!ps)
		exit(1);

	for (i=0; i<numRestantes; i++) {
		ps[i] = retornaTopo(h);
		tirarElemento(h);
	}

	for (i=0; i<numRestantes; i++) { 
		mudaPrioridade(ps[i]);
		adicionarElemento(h, ps[i]);
	}

	free(ps);
}
